import React from 'react'
import { Tag as AntTag } from 'antd'

export interface TagProps extends React.ComponentProps<typeof AntTag> {
  /**
   * 标签是否可以关闭（点击默认关闭）
   * @default false
   */
  closable?: boolean
  /**
   * 自定义关闭按钮
   */
  closeIcon?: React.ReactNode
  /**
   * 标签色
   */
  color?: string
  /**
   * 设置图标
   */
  icon?: React.ReactNode
  /**
   * 关闭时的回调（可通过 e.preventDefault() 来阻止默认行为）
   */
  onClose?: (e: React.MouseEvent<HTMLElement, MouseEvent>) => void
}

export const Tag = ({ children, ...restProps }: TagProps) => <AntTag {...restProps}>{children}</AntTag>

export * from './custom'
